// 版权所有2009 Go作者。版权所有。
// 此源代码的使用受BSD样式
// 许可证的约束，该许可证可以在许可证文件中找到。

package runtime

import (
	"runtime/internal/atomic"
	"unsafe"
)

// 这些函数不能有go:noescape注释，因为虽然ptr不能转义，但new可以转义。
// 如果new被标记为not escape，编译器将对存储的指针值做出错误的
// escape分析决定。

// atomicwb在原子指针写入之前执行写入屏障。
// 调用方应使用“if writeBarrier.enabled”保护调用。
// 
// go:nosplit 
func atomicwb(ptr *unsafe.Pointer, new unsafe.Pointer) {
	slot := (*uintptr)(unsafe.Pointer(ptr))
	if !getg().m.p.ptr().wbBuf.putFast(*slot, uintptr(new)) {
		wbBufFlush(slot, uintptr(new))
	}
}

// atomicstorep以原子方式执行*ptr=new并调用写屏障。
// 
// go:nosplit 
func atomicstorep(ptr unsafe.Pointer, new unsafe.Pointer) {
	if writeBarrier.enabled {
		atomicwb((*unsafe.Pointer)(ptr), new)
	}
	atomic.StorepNoWB(noescape(ptr), new)
}

// 与上述类似，但根据同步/原子的uintptr操作实现。
// 我们不能只调用运行时例程，因为竞赛检测器希望
// 能够截获同步/原子表单，但不能截获运行时表单。

// go:linkname sync_atomic_StoreUintptr sync/atomic.StoreUintptr 
func sync_atomic_StoreUintptr(ptr *uintptr, new uintptr)

// go:linkname sync_atomic_StorePointer sync/atomic.StorePointer 
// go nosplit 
func sync_atomic_StorePointer(ptr *unsafe.Pointer, new unsafe.Pointer) {
	if writeBarrier.enabled {
		atomicwb(ptr, new)
	}
	sync_atomic_StoreUintptr((*uintptr)(unsafe.Pointer(ptr)), uintptr(new))
}

// go:linkname sync_atomic_SwapUintptr 
func sync_atomic_SwapUintptr(ptr *uintptr, new uintptr) uintptr

// go
// go:nosplit 
func sync_atomic_SwapPointer(ptr *unsafe.Pointer, new unsafe.Pointer) unsafe.Pointer {
	if writeBarrier.enabled {
		atomicwb(ptr, new)
	}
	old := unsafe.Pointer(sync_atomic_SwapUintptr((*uintptr)(noescape(unsafe.Pointer(ptr))), uintptr(new)))
	return old
}

// go:linkname sync\u atomic\u CompareandWapuIntPtr sync/atomic.CompareandWapuIntPtr 
func sync_atomic_CompareAndSwapUintptr(ptr *uintptr, old, new uintptr) bool

// go:linkname sync\u atomic\u CompareandWappointer sync/atomic.CompareandWappointer 
// go:nosplit 
func sync_atomic_CompareAndSwapPointer(ptr *unsafe.Pointer, old, new unsafe.Pointer) bool {
	if writeBarrier.enabled {
		atomicwb(ptr, new)
	}
	return sync_atomic_CompareAndSwapUintptr((*uintptr)(noescape(unsafe.Pointer(ptr))), uintptr(old), uintptr(new))
}
